package cn.edu.njust.alg.leetcode;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * 创建时间：2015年6月15日 上午11:47:37 项目名称：AlgorithmExercise
 * 
 * @author Zhou Xiumin
 * @version 1.0
 * @since JDK 1.7.0_21 文件名称：Leetcode_0_t.java 类说明：
 */
public class Leetcode_217_contains_duplicate {

	public static boolean containsDuplicate(int[] nums) {

		Set<Integer> set = new HashSet<Integer>();
		for (int n : nums) {
			if (set.contains(n)) {
				return true;
			} else {
				set.add(n);
			}
		}
		return false;
	}

	public static boolean containsNearbyDuplicate(int[] nums, int k) {
		Map<Integer, Integer> map = new HashMap<Integer, Integer>();
		for (int i = 0; i < nums.length; i++) {
			if (map.containsKey(nums[i]) && i - map.get(nums[i]) <= k) {
				return true;
			} else {
				map.put(nums[i], i);
			}

		}

		return false;
	}

	public static void main(String[] args) {
		int[] nums = { 1, 2, 34, 5, 1 };
		System.out.println(containsDuplicate(nums));
		System.out.println(containsNearbyDuplicate(nums, 5));
	}

}
